from __future__ import annotations

from typing import Iterator

import httpx
# imports
import pytest  # used for our unit tests
from openai._legacy_response import HttpxBinaryResponseContent

# unit tests

# Helper function to create an httpx.Response object with given content and encoding
def make_response(content: bytes, encoding: str = "utf-8") -> httpx.Response:
    # Use httpx.Response directly with content and encoding
    return httpx.Response(200, content=content, headers={"Content-Type": f"text/plain; charset={encoding}"})

# ------------------------
# 1. Basic Test Cases
# ------------------------


















#------------------------------------------------
from typing import Iterator

# imports
import pytest
from openai._legacy_response import HttpxBinaryResponseContent


# function to test
class DummyResponse:
    """
    A dummy httpx.Response-like class for testing.
    """
    def __init__(self, text: str):
        self._text = text

    def iter_text(self, chunk_size: int | None = None) -> Iterator[str]:
        """
        Mimics httpx.Response.iter_text. Yields text in chunks of chunk_size.
        If chunk_size is None, yields the whole text at once.
        """
        if chunk_size is None:
            yield self._text
        else:
            if not isinstance(chunk_size, int) or chunk_size <= 0:
                raise ValueError("chunk_size must be a positive integer or None")
            for i in range(0, len(self._text), chunk_size):
                yield self._text[i:i+chunk_size]
from openai._legacy_response import HttpxBinaryResponseContent

# unit tests

# ---------------- Basic Test Cases ----------------